from langchain.prompts import ChatPromptTemplate
from tool import langchain_client
if __name__ == '__main__':
    customer_review = """
    这款吹叶机非常神奇。 它有四个设置:\
    吹蜡烛、微风、风城、龙卷风。\
    两天后就到了，正好赶上我妻子的\
    周年纪念礼物。 \
    我想我的妻子会喜欢它到说不出话来。 \
    到目前为止，我是唯一一个使用它的人，而且我一直\
    每隔一天早上用它来清理草坪上的叶子。 \
    它比其他吹叶机稍微贵一点，\
    但我认为它的额外功能是值得的。
    """
    review_template = """
    对于以下文本，请从中提取以下信息:
    
    礼物:该商品是作为礼物送给别人的吗? \
    如果是，则回答 是的;如果否或未知，则回答 不是。
    
    交货天数:产品需要多少天\
    到达? 如果没有找到该信息，则输出-1。
    
    价钱:提取有关价值或价格的任何句子，\
    并将它们输出为逗号分隔的 Python 列表。
    使用以下键将输出格式化为 JSON: 
    礼物
    交货天数
    价钱
    
    文本: {text}
    """
    prompt_template = ChatPromptTemplate.from_template(review_template)

    print("提示模版:", prompt_template)
    messages = prompt_template.format_messages(text=customer_review)

    response = langchain_client.invoke(messages)
    print("结果类型:", type(response.content))
    print("结果:", response.content)

